accelerate
diffusers==0.19.3
transformers
omegaconf
opencv-python
six
einops
IPython